package com.yizhiliulianta.wx.controller;

import com.yizhiliulianta.wx.handler.WxChatMsgFactory;
import com.yizhiliulianta.wx.handler.WxChatMsgHandler;
import com.yizhiliulianta.wx.utils.MessageUtil;
import com.yizhiliulianta.wx.utils.SHA1;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.Map;
import java.util.Objects;

@RestController
@Slf4j
public class CallBackController {

    private static final String token = "yizhiliulianta";

    @Resource
    private WxChatMsgFactory wxChatMsgFactory;


    @GetMapping("callback")
    public String callback(@RequestParam("signature") String signature,
                           @RequestParam("timestamp") String timestamp,
                           @RequestParam("nonce") String nonce,
                           @RequestParam("echostr") String echostr) {
        log.info("get验签请求参数：signature:{}，timestamp:{}，nonce:{}，echostr:{}",
                signature, timestamp, nonce, echostr);
        String shaStr = SHA1.getSHA1(token, timestamp, nonce, "");
        if (signature.equals(shaStr)) {
            return echostr;
        }
        return "unknown";
    }

    @PostMapping(value = "callback", produces = "application/xml;charset=UTF-8")
    public String callback(
            @RequestBody String requestBody,
            @RequestParam("signature") String signature,
            @RequestParam("timestamp") String timestamp,
            @RequestParam("nonce") String nonce,
            @RequestParam(value = "echostr",required = false) String echostr) {
        log.info("接收到微信的请求：requestBody:{},signature:{},timestamp:{},nonce:{},echostr:{}"
                ,requestBody,signature,timestamp,nonce,echostr);
        Map<String, String> msgMap = MessageUtil.parseXml(requestBody);
        String msgType = msgMap.get("MsgType");
        String event = msgMap.get("Event");
        log.info("msgType:{},event:{}",msgType,event);

        StringBuilder sb = new StringBuilder();
        sb.append(msgType);
        if (event!=null){
            sb.append(".");
            sb.append(event);
        }
        String msgTypeKey = sb.toString();

        WxChatMsgHandler wxChatMsgHandler = wxChatMsgFactory.getHandlerByMsgType(msgTypeKey);

        if (Objects.isNull(wxChatMsgHandler)){
            return "unknown";
        }

        String replyContent = wxChatMsgHandler.dealMsg(msgMap);
        log.info("replyContent:{}",replyContent);
        return replyContent;
    }
}
